#!/usr/bin/python3

# 1. 导入包
import pymysql

# 2. 建立 数据库连接
# 调用env.py文件中的变量连接数据库
from env import MYSQL_HOST
from env import MYSQL_DATABASE
from env import MYSQL_PASSWORD
from env import MYSQL_USERNAME

MYSQL_HOST = MYSQL_HOST
MYSQL_DATABASE = MYSQL_DATABASE
MYSQL_USERNAME = MYSQL_USERNAME
MYSQL_PASSWORD = MYSQL_PASSWORD

db = pymysql.connect(host=MYSQL_HOST,
                     user=MYSQL_USERNAME,
                     password=MYSQL_PASSWORD,
                     database=MYSQL_DATABASE)

# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()

# 使用 execute()  方法执行 SQL 查询
cursor.execute("select * from user limit 1")

# 使用 fetchone() 方法获取单条数据.
# 使用 fetchall() 方法获取全部数据.
data = cursor.fetchall()

print(data)

# 关闭数据库连接
db.close()

